# 模块：写入 xlsx 文件
from openpyxl import Workbook
WB = None
WS = None

def initWorkBook(sheet_name='sheet1',**kwargs):
    # 初始化工作簿和工作表,定义工作表和列名
    # 使用示例：init('工作表1'，A='姓名',B='年龄',.....)  列大小写均可
    global WB
    global WS 

    WB = Workbook()
    WS = WB.active
    WS.title = sheet_name

    print(kwargs)
    for key,value in kwargs.items():
        print(type(key))
        key:str
        value:str
        WS[key+'1'] = value




def WriteToExcelFile(**kwargs):
    #自动向excel表的最后一个空白行写入数据
    #使用示例： WriteToExcelFile(a='张三',b='20')
    global WB
    global WS
    #获取列中有数据的最后一行

    last_row = WS.max_row
    tag_row = last_row + 1

    for key,value in kwargs.items():
      WS[f"{key}{tag_row}"] = value



def SaveExcel(wbPath):
    # 接收文件路径，保存文件
    # 使用示例：SaveExcel('D://test.xlsx')
    global WB
    WB.save(wbPath)
    
# 修改列宽
def Modify_ColumnWidth(col,col_width):
    # 使用示例：Modify_ColumnWidth('A',30)
    col:str
    col_width:float
    WS.column_dimensions[col].width = col_width

if __name__ == "__main__":
    print('this is outToExcel_Module.py')

